Parser States Already Know: Structure-Conditioned KV Persistence for Structured Generation
解析器状态指导的结构化生成 KV 缓存持久化
🧭 LLM 推理优化 · 结构化生成 · KV 压缩 ★ 深度解读
Hangzhou Institute for Advanced Study, University of Chinese Academy of Sciences
利用约束解码的解析器状态设计离线校准的 KV 持久化策略,大幅提升结构化生成在压缩推理下的可靠性
📌 工作参考价值(4 点)
- 电商 Agent 中大量 JSON/function call 输出,可直接复用 constrained decoding 暴露的 parser states 作为 KV 保留信号。例如将必需字段、枚举值、参数名标记为高保护 bucket,避免压缩导致关键 schema 信息丢失,降低下游调用失败率。
- 借鉴「离线校准 + 线上查表」的工程化方式:用开发集离线测量每个结构标签+层组的任务错误敏感性和注意力失真,生成策略表;线上只做轻量查找和常量预算状态更新,适合低延迟、高并发推荐/搜索 Agent 服务。
- 任务错误敏感性设置保护下限比单纯依赖注意力分数更可靠。在推荐理由生成、商品参数抽取等结构化任务中,优先保护影响最终可执行性的 token(如字段名、枚举值),可以显著提升压缩后的可靠性。
- 分层分组思路:同一结构角色在不同 Transformer 层采用不同精度(低层低精度、高层高精度),可迁移到长上下文推荐模型或多轮 Agent 对话的 KV 管理,进一步节省显存。